class Solution {
private:
    typedef pair<string, int> PSI;
    struct cmp
    {
        bool operator()(const PSI& a, const PSI& b)
        {
            if(a.second == b.second)    // 频次相同，字典序按照大堆的方式排列
            {
                return a.first < b.first;
            }
            return a.second > b.second;
        }
    };
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        // 1. 统计一下每一个单词的频次
        unordered_map<string, int> hash;
        for(auto& s : words)
            hash[s]++;

        // 2. 创建一个大小为 k 的堆
        priority_queue<PSI, vector<PSI>, cmp> heap;

        // 3. TopK 的主逻辑
        for(auto& psi : hash)
        {
            heap.push(psi);
            if(heap.size() > k)
                heap.pop();
        }

        // 4. 提取结果
        vector<string> ret(k);
        for(int i = k - 1; i >= 0; i--)
        {
            ret[i] = heap.top().first;
            heap.pop();
        }
        return ret;
    }
};